# 导入YOLO模型
from ultralytics import YOLO

# 加载模型
model = YOLO('yolov8m.pt')  # 加载预训练模型（推荐用于训练）

# 训练模型
model.train(
    data='plastic.yaml',  # 数据集配置文件
    epochs=20,  # 训练轮数
    imgsz=(1280, 720),  # 图像尺寸
    batch=4,  # 批次大小
    optimizer="Adam",  # 优化器
    lr0=1e-3,  # 初始学习率
)
